package net.liuyimin.service;

import net.liuyimin.util.DistributedLockClient;
import net.liuyimin.util.DistributedRedisLock;
import org.apache.commons.lang3.StringUtils;
import org.redisson.api.RLock;
import org.redisson.api.RedissonClient;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.Arrays;
import java.util.UUID;
import java.util.concurrent.TimeUnit;

@Service
public class StockService {

    @Resource
    private StringRedisTemplate stringRedisTemplate;

    @Resource
    private DistributedLockClient distributedLockClient;

    @Resource
    private RedissonClient redissonClient;

    /**
     * 基于Redis实现分布式锁: 演示高并发下的超卖问题
     */
    public void deduct() {
        // 1.查询库存信息
        String stock = this.stringRedisTemplate.opsForValue().get("stock").toString();
        // 2.判断库存是否充足
        if (stock != null && stock.length() != 0) {
            Integer st = Integer.valueOf(stock);
            if (st > 0) {
                // 3.扣减库存
                this.stringRedisTemplate.opsForValue().set("stock", String.valueOf(--st));
            }
        }
    }

    /**
     * 基于Redis实现分布式锁: 基本实现分布式锁
     */
    public void deduct2() {
        // 加锁 setnx
        Boolean lock = this.stringRedisTemplate.opsForValue().setIfAbsent("lock", "lock");

        // 重试: 递归调用
        if (!lock) {
            try {
                // 此处睡眠50毫秒(根据电脑性能)，再次尝试加锁，直到成功为止. 作用是避免死锁,防止栈内存溢出.
                Thread.sleep(50);

                this.deduct2();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        } else {
            try {
                // 1.查询库存信息
                String stock = this.stringRedisTemplate.opsForValue().get("stock").toString();
                // 2.判断库存是否充足
                if (stock != null && stock.length() != 0) {
                    Integer st = Integer.valueOf(stock);
                    if (st > 0) {
                        // 3.扣减库存
                        this.stringRedisTemplate.opsForValue().set("stock", String.valueOf(--st));
                    }
                }
            } finally {
                // 解锁 del
                this.stringRedisTemplate.delete("lock");
            }
        }
    }

    /**
     * 基于Redis实现分布式锁: 优化重试操作步骤,将递归调用换成循环
     */
    public void deduct3() {
        // 加锁 setnx
        while (!this.stringRedisTemplate.opsForValue().setIfAbsent("lock", "lock")) {
            // 重试: 循环
            try {
                // 此处睡眠可以使得竞争压力变小,本机可以认为是CAS来实现的(不停地自旋去实现获取锁)
                Thread.sleep(50);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
        try {
            // 1.查询库存信息
            String stock = this.stringRedisTemplate.opsForValue().get("stock").toString();
            // 2.判断库存是否充足
            if (stock != null && stock.length() != 0) {
                Integer st = Integer.valueOf(stock);
                if (st > 0) {
                    // 3.扣减库存
                    this.stringRedisTemplate.opsForValue().set("stock", String.valueOf(--st));
                }
            }
        } finally {
            // 解锁 del
            this.stringRedisTemplate.delete("lock");
        }
    }

    /**
     * 基于Redis实现分布式锁: 防死锁+防误删
     */
    public void deduct4() {
        // 防误删的唯一标识
        String uuid = UUID.randomUUID().toString();
        // 加锁 setnx + 防死锁
        while (!this.stringRedisTemplate.opsForValue().setIfAbsent("lock", uuid, 3, TimeUnit.SECONDS)) {
            // 重试: 循环
            try {
                // 此处睡眠可以使得竞争压力变小,本机可以认为是CAS来实现的(不停地自旋去实现获取锁)
                Thread.sleep(50);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
        try {
            // 1.查询库存信息
            String stock = this.stringRedisTemplate.opsForValue().get("stock").toString();
            // 2.判断库存是否充足
            if (stock != null && stock.length() != 0) {
                Integer st = Integer.valueOf(stock);
                if (st > 0) {
                    // 3.扣减库存
                    this.stringRedisTemplate.opsForValue().set("stock", String.valueOf(--st));
                }
            }
        } finally {
            // 防误删 : 先判断是否是自己的锁，再解锁
            if (StringUtils.equals(this.stringRedisTemplate.opsForValue().get("lock"), uuid)) {
                // 解锁 del
                this.stringRedisTemplate.delete("lock");
            }
        }
    }

    /**
     * 基于Redis实现分布式锁: 使用Lua脚本解决防误删除的原子性问题
     */
    public void deduct5() {
        // 防误删的唯一标识
        String uuid = UUID.randomUUID().toString();
        // 加锁 setnx + 防死锁
        while (!this.stringRedisTemplate.opsForValue().setIfAbsent("lock", uuid, 3, TimeUnit.SECONDS)) {
            // 重试: 循环
            try {
                // 此处睡眠可以使得竞争压力变小,本机可以认为是CAS来实现的(不停地自旋去实现获取锁)
                Thread.sleep(50);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
        try {
            // 1.查询库存信息
            String stock = this.stringRedisTemplate.opsForValue().get("stock").toString();
            // 2.判断库存是否充足
            if (stock != null && stock.length() != 0) {
                Integer st = Integer.valueOf(stock);
                if (st > 0) {
                    // 3.扣减库存
                    this.stringRedisTemplate.opsForValue().set("stock", String.valueOf(--st));
                }
            }
        } finally {
            // 先判断是否自己的锁，再解锁
            String script = "if redis.call('get', KEYS[1]) == ARGV[1] " +
                    "then " +
                    "   return redis.call('del', KEYS[1]) " +
                    "else " +
                    "   return 0 " +
                    "end";
            this.stringRedisTemplate.execute(new DefaultRedisScript<>(script, Boolean.class), Arrays.asList("lock"), uuid);
        }
    }

    /**
     * 基于Redis实现分布式锁: 实现可重入锁
     */
    public void deduct6() {
        // 可重入锁
        DistributedRedisLock redisLock = this.distributedLockClient.getRedisLock("lock:stock");
        redisLock.lock();

        try {
            // 1.查询库存信息
            String stock = this.stringRedisTemplate.opsForValue().get("stock").toString();
            // 2.判断库存是否充足
            if (stock != null && stock.length() != 0) {
                Integer st = Integer.valueOf(stock);
                if (st > 0) {
                    // 3.扣减库存
                    this.stringRedisTemplate.opsForValue().set("stock", String.valueOf(--st));
                }
            }
        } finally {
            // 解锁
            redisLock.unlock();
        }
    }

    /**
     * 基于Redis实现分布式锁: 借助 Redisson 快速实现分布式锁
     */
    public void deduct7() {
        // 获取锁
        RLock lock = this.redissonClient.getLock("lock");
        lock.lock();
        try {
            // 1.查询库存信息
            String stock = this.stringRedisTemplate.opsForValue().get("stock").toString();
            // 2.判断库存是否充足
            if (stock != null && stock.length() != 0) {
                Integer st = Integer.valueOf(stock);
                if (st > 0) {
                    // 3.扣减库存
                    this.stringRedisTemplate.opsForValue().set("stock", String.valueOf(--st));
                }
            }
        } finally {
            // 释放锁
            lock.unlock();
        }
    }

}
